// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_BLOB_READER_H_
#define CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_BLOB_READER_H_

#include <memory>

#include "content/browser/service_worker/service_worker_url_request_job.h"
#include "net/base/io_buffer.h"
#include "net/url_request/url_request.h"

namespace content {

// Reads a blob response for ServiceWorkerURLRequestJob.
// Owned by ServiceWorkerURLRequestJob.
class ServiceWorkerBlobReader : public net::URLRequest::Delegate {
public:
    explicit ServiceWorkerBlobReader(ServiceWorkerURLRequestJob* owner);
    ~ServiceWorkerBlobReader() override;

    // Starts reading the blob. owner_->OnResponseStarted will be called when the
    // response starts.
    void Start(std::unique_ptr<storage::BlobDataHandle> blob_data_handle,
        const net::URLRequestContext* request_context);

    scoped_refptr<net::IOBufferWithSize> response_metadata()
    {
        return blob_request_->response_info().metadata;
    }

    // Same as URLRequestJob::ReadRawData. If ERR_IO_PENDING is returned,
    // owner_->OnReadRawDataComplete will be called when the read completes.
    int ReadRawData(net::IOBuffer* buf, int buf_size);

    // net::URLRequest::Delegate overrides
    void OnReceivedRedirect(net::URLRequest* request,
        const net::RedirectInfo& redirect_info,
        bool* defer_redirect) override;
    void OnAuthRequired(net::URLRequest* request,
        net::AuthChallengeInfo* auth_info) override;
    void OnCertificateRequested(
        net::URLRequest* request,
        net::SSLCertRequestInfo* cert_request_info) override;
    void OnSSLCertificateError(net::URLRequest* request,
        const net::SSLInfo& ssl_info,
        bool fatal) override;
    void OnResponseStarted(net::URLRequest* request) override;
    void OnReadCompleted(net::URLRequest* request, int bytes_read) override;

private:
    ServiceWorkerURLRequestJob* owner_;
    std::unique_ptr<net::URLRequest> blob_request_;

    DISALLOW_COPY_AND_ASSIGN(ServiceWorkerBlobReader);
};

} // namespace content

#endif // CONTENT_BROWSER_SERVICE_WORKER_SERVICE_WORKER_BLOB_READER_H_
